package com.example.lettcode._20241014;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;

/*
57. 插入区间
给你一个 无重叠的 ，按照区间起始端点排序的区间列表 intervals，其中 intervals[i] = [starti, endi] 表示第 i 个区间的开始和结束，并且 intervals 按照 starti 升序排列。同样给定一个区间 newInterval = [start, end] 表示另一个区间的开始和结束。
在 intervals 中插入区间 newInterval，使得 intervals 依然按照 starti 升序排列，且区间之间不重叠（如果有必要的话，可以合并区间）。
返回插入之后的 intervals。
注意 你不需要原地修改 intervals。你可以创建一个新数组然后返回它。

示例 1：
输入：intervals = [[1,3],[6,9]], newInterval = [2,5]
输出：[[1,5],[6,9]]

示例 2：
输入：intervals = [[1,2],[3,5],[6,7],[8,10],[12,16]], newInterval = [4,8]
输出：[[1,2],[3,10],[12,16]]
解释：这是因为新的区间 [4,8] 与 [3,5],[6,7],[8,10] 重叠。
 */
public class Cha_ru_qu_jian {

    public static void main(String[] args) {
        int[][] intervals = {{1,3},{6,9}};
        int[] newInterval = {2,5};
        int[][] insert = insert3(intervals, newInterval);
        for (int i = 0; i < insert.length; i++) {
            for (int j = 0; j < insert[i].length; j++) {
                System.out.print(insert[i][j] +  " ");
            }
            System.out.println();
        }
    }

    public static int[][] insert3(int[][] intervals, int[] newInterval) {
        List<int[]> list = new ArrayList<>();
        int n = intervals.length;
        int i = 0;
        // 小于newInterval的数据
        while(i < n && intervals[i][1] < newInterval[0]) {
            list.add(intervals[i]);
            i++;
        }

        // 重合的数据
        while(i < n && intervals[i][0] <= newInterval[1]) {
            newInterval[0] = Math.min(intervals[i][0], newInterval[0]);
            newInterval[1] = Math.max(intervals[i][1], newInterval[1]);
            i++;
        }
        list.add(newInterval);

        // 大于的数据
        while (i < n && intervals[i][0] > newInterval[1]) {
            list.add(intervals[i]);
            i++;
        }
        return list.toArray(new int[list.size()][]);
    }



    public static int[][] insert2(int[][] intervals, int[] newInterval) {
        List<int[]> list = new ArrayList<>();
        int n = intervals.length;
        int i = 0;
        // 小于newInterval区间的数据
        while (i < n && intervals[i][1] < newInterval[0]) {
            list.add(intervals[i]);
            i++;
        }

        // 在区间范围内的
        while(i < n && intervals[i][0] <= newInterval[1]) {
            newInterval[0] = Math.min(intervals[i][0], newInterval[0]);
            newInterval[1] = Math.max(intervals[i][1], newInterval[1]);
            i++;
        }
        list.add(newInterval);

        while(i < n && intervals[i][0] > newInterval[1]) {
            list.add(intervals[i]);
            i++;
        }
        return list.toArray(new int[list.size()][]);
    }


    public static int[][] insert(int[][] intervals, int[] newInterval) {
        int[][] newInt = new int[intervals.length + 1][2];
        for (int i = 0; i < intervals.length; i++) {
            newInt[i] = intervals[i];
        }
        newInt[intervals.length] = newInterval;

        Arrays.sort(newInt, new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                return o1[0] - o2[0];
            }
        });

        int[] prev = newInt[0];
        int[] cur;
        int k = 0;
        for (int i = 1; i < newInt.length; i++) {
            cur = newInt[i];
            if (prev[1] >= cur[0]) {
                int e = Math.max(prev[1], cur[1]);
                prev[1] = e;
            } else {
                newInt[k++] = prev;
                prev = cur;
            }
        }
        newInt[k++] = prev;
        return Arrays.copyOfRange(newInt, 0, k);
    }
}
